hackerrank matrix script problem solution in python programming